python3 imitate_episodes.py --task_name real_transfer_cube --ckpt_dir ckpt_diffusion --policy_class Diffusion --batch_size 4 --lr 1e-5 --seed 0 --num_steps 100000 
